import pygame
from pygame.locals import*
import random

class Parede(pygame.sprite.Sprite):
    def __init__(self, x, y, img):
        pygame.sprite.Sprite.__init__(self)
        self.image  = pygame.image.load(img)
        self.rect = self.image.get_rect()
        self.rect.top = y
        self.rect.left = x
        self.tipo = 'parede'

    def cria_fases(self):
        linhas_fase = open('fases/fase.txt').readlines()
        fase = open('fases/fase.txt', 'w')

        for linha in linhas_fase:
            nova_linha = []
            for caractere in linha:
                if caractere == '=' or caractere == ' ':
                    sorteio = random.randint(0,1)
                    if sorteio == 0:
                        caractere = '='
                    else:
                        caractere = ' '
                nova_linha.append(caractere)
            fase.write(''.join(nova_linha))

        fase.close()

class Muro(pygame.sprite.Sprite):
    def __init__(self, x, y, img):
        pygame.sprite.Sprite.__init__(self)
        self.image  = pygame.image.load(img)
        self.rect = self.image.get_rect()
        self.rect.top = y
        self.rect.left = x
        self.tipo = 'muro_destroi'
        lista_itens = ['aumenta_vida', '', '', '', 'chuta_bombas', '', '', '', 'chuta_bombas', '', '', '', 'bomba_relogio', '', '', '', 'bomba_relogio', '', '', '', 'aumenta_bomba', '', '', '', 'aumenta_bomba', '', '', '', 'aumenta_bomba', '', '', '', 'aumenta_bomba', '', '', '', 'aumenta_fogo', '', '', '', 'aumenta_fogo', '', '', '', 'aumenta_fogo', '', '', '', 'aumenta_fogo', '', '', '', 'aumenta_fogo', '', '', '', 'aumenta_velocidade', '', '', '', 'aumenta_velocidade']
        self.item_tipo = random.choice(lista_itens)

class Itens(pygame.sprite.Sprite):
    def __init__(self, parede, grupoItens):
        pygame.sprite.Sprite.__init__(self)
        self.tipo = parede.item_tipo
        if parede.item_tipo == 'aumenta_bomba':
            self.image  = pygame.image.load('imagens/aumenta_bomba.png')
        elif parede.item_tipo == 'aumenta_fogo':
            self.image = pygame.image.load('imagens/aumenta_fogo.png')
        elif parede.item_tipo == 'aumenta_velocidade':
            self.image = pygame.image.load('imagens/aumenta_velocidade.png')
        elif parede.item_tipo == 'bomba_relogio':
            self.image = pygame.image.load('imagens/bomba_relogio.png')
        elif parede.item_tipo == 'chuta_bombas':
            self.image = pygame.image.load('imagens/chuta_bombas.png')
        elif parede.item_tipo == 'aumenta_vida':
            self.image = pygame.image.load('imagens/aumenta_vida.png')
        self.rect = self.image.get_rect()
        self.rect.top = parede.rect.top
        self.rect.left = parede.rect.left
        grupoItens.add(self)
